package cn.wgx.common.security.service;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.oauth2.common.OAuth2AccessToken;
import org.springframework.security.oauth2.common.OAuth2RefreshToken;
import org.springframework.security.oauth2.provider.token.TokenStore;
import org.springframework.stereotype.Service;

@Service
public class AuthService {

    @Autowired
    TokenStore tokenStore;

    public boolean removeToken(String token){
        OAuth2AccessToken oAuth2AccessToken = tokenStore.readAccessToken(token);
        if(oAuth2AccessToken != null){
            OAuth2RefreshToken refreshToken = oAuth2AccessToken.getRefreshToken();
            tokenStore.removeAccessToken(oAuth2AccessToken);
            if(refreshToken != null){
                tokenStore.removeRefreshToken(refreshToken);
            }

        }
        return true;
    }

}
